// Just some shapes.
//
// Created on: 2017/5/13
//     Author: Bochen Yan

#pragma once

#include <iostream>
#include <string>
using namespace std;

class shape{
public:
    shape(const string);
    string getColor() const;
private:
    string color;
};

class rectangle: public shape{
public:
    rectangle(const string);
    rectangle(const string, int, int);
    int getArea() const;
protected:
    int width;
    int height;
};

class square: public rectangle{
public:
    square(const string, int);
    int getArea() const;
protected:
    int length;
};